import os
from openpyxl import Workbook


def list_files_in_folder(folder_path):
    # 获取文件夹中的文件名
    file_names = os.listdir(folder_path)
    return file_names


def write_to_excel(file_names, excel_file):
    # 创建一个新的Excel工作簿
    wb = Workbook()
    # 选择第一个工作表
    ws = wb.active
    # 在第一列写入文件名
    for index, file_name in enumerate(file_names, start=1):
        file_name = file_name.split('.')[0]
        ws.cell(row=index, column=1, value=file_name)
    # 保存Excel文件
    wb.save(excel_file)


if __name__ == "__main__":
    folder_path = "/Users/lidaoyu/Downloads/简历/"  # 替换为你的文件夹路径
    directory = os.path.dirname(folder_path)  # 文件夹所在路径
    excel_file = f'{directory}/简历姓名.xlsx'  # 输出的Excel文件名

    file_names = list_files_in_folder(folder_path)
    write_to_excel(file_names, excel_file)
